#!/bin/bash

# Tiny QA - 模型设置脚本

echo "========================================"
echo "Tiny QA - AI模型设置向导"
echo "========================================"

# 创建模型目录
mkdir -p models

echo ""
echo "🤖 请选择您需要的AI模型:"
echo ""
echo "1. 轻量级模型 (推荐新手)"
echo "   - qwen2.5:3b (3GB, 速度快，适合入门)"
echo "   - nomic-embed-text (嵌入模型)"
echo ""
echo "2. 标准模型 (推荐)"
echo "   - qwen2.5:7b (7GB, 平衡性能和效果)"
echo "   - nomic-embed-text (嵌入模型)"
echo ""
echo "3. 高性能模型 (需要更多资源)"
echo "   - qwen2.5:14b (14GB, 更好的效果)"
echo "   - nomic-embed-text (嵌入模型)"
echo ""
echo "4. 自定义模型"
echo "   - 手动指定模型名称"
echo ""

read -p "请选择 (1-4): " choice

case $choice in
    1)
        echo "📦 选择轻量级模型..."
        cat > models/model_config.txt << EOF
# 轻量级模型配置
CHAT_MODEL=qwen2.5:3b
EMBED_MODEL=nomic-embed-text
EOF
        ;;
    2)
        echo "📦 选择标准模型..."
        cat > models/model_config.txt << EOF
# 标准模型配置
CHAT_MODEL=qwen2.5:7b
EMBED_MODEL=nomic-embed-text
EOF
        ;;
    3)
        echo "📦 选择高性能模型..."
        cat > models/model_config.txt << EOF
# 高性能模型配置
CHAT_MODEL=qwen2.5:14b
EMBED_MODEL=nomic-embed-text
EOF
        ;;
    4)
        echo "📝 自定义模型配置..."
        read -p "请输入对话模型名称 (如: qwen2.5:7b): " chat_model
        read -p "请输入嵌入模型名称 (如: nomic-embed-text): " embed_model
        
        cat > models/model_config.txt << EOF
# 自定义模型配置
CHAT_MODEL=${chat_model}
EMBED_MODEL=${embed_model}
EOF
        ;;
    *)
        echo "❌ 无效选择，使用默认配置"
        cat > models/model_config.txt << EOF
# 默认模型配置
CHAT_MODEL=qwen2.5:7b
EMBED_MODEL=nomic-embed-text
EOF
        ;;
esac

echo ""
echo "✅ 模型配置已保存到 models/model_config.txt"
echo ""
echo "📋 配置内容:"
cat models/model_config.txt

echo ""
echo "🚀 下一步:"
echo "1. 运行 ./scripts/docker-deploy.sh 启动服务"
echo "2. 系统将自动下载您选择的模型"
echo "3. 首次启动可能需要10-30分钟下载模型"
echo ""
echo "💡 提示: 模型下载完成后，后续启动会很快" 